package com.tangsm.spring.boot.csv;

import org.apache.commons.net.ftp.FTPClient;
import org.apache.commons.net.ftp.FTPFile;
import org.apache.commons.net.ftp.FTPReply;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;

/**
 * Ftp工具类
 */
public class FtpUtils {

    public static void main(String[] args) throws Exception {

//        // =======上传测试============
        String localFile = "C:\\Users\\bhlgo\\Downloads\\中国.txt";

        String url = "ip地址";
        int port = 23;
        String username = "账号";
        String password = "密码";
        String path = "目录";
        String filename = "中国.txt";
        InputStream input = Files.newInputStream(new File(localFile).toPath());

        uploadFile(url, port, username, password, path, filename, input);
//        // =======上传测试============

        // =======下载测试============
        String localPath = "D:\\";
        url = "192.168.120.133";
        port = 21;
        username ="ftpuser";
        password ="ftpuser";
        String remotePath ="/data/admftp/kw";
        String fileName = "test.txt";
        downFile(url, port, username, password, remotePath, fileName, localPath);
        // =======下载测试============

    }

    /**
     * Description: 向FTP服务器上传文件
     *
     * @param url      FTP服务器 hostname
     * @param port     FTP服务器端口 默认端 21
     * @param username FTP登录账号
     * @param password FTP登录密码
     * @param path     FTP服务器保存目录
     * @param filename 上传到FTP服务器上的文件名
     * @param input    输入流
     * @return 成功返回true，否则返回false
     */
    public static boolean uploadFile(String url, int port, String username, String password, String path,
                                     String filename, InputStream input) {
        boolean success = false;
        FTPClient ftp = new FTPClient();
        try {
            int reply;
            // 连接初始化
            ftp.connect(url, port);
            // 登录
            // 如果采用默认端口，可以使用ftp.connect(url)的方式直接连接FTP服务器
            ftp.login(username, password);
//            ftp.setFileType(FTPClient.BINARY_FILE_TYPE);
//            ftp.setControlEncoding("GBK");
            reply = ftp.getReplyCode();
            if (!FTPReply.isPositiveCompletion(reply)) {
                ftp.disconnect();
                return success;
            }
            ftp.changeWorkingDirectory(path);
            // 设置文件名上传的编码格式为 utf-8
            ftp.storeFile(new String(filename.getBytes(StandardCharsets.UTF_8), StandardCharsets.ISO_8859_1), input);

            input.close();
            ftp.logout();
            success = true;
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            if (ftp.isConnected()) {
                try {
                    ftp.disconnect();
                } catch (IOException ioe) {
                    ioe.printStackTrace();
                }
            }
        }
        return success;
    }

    /**
     * 从FTP服务器下载文件
     *
     * @param url        FTP服务器hostname
     * @param port       FTP服务器端口
     * @param username   FTP登录账号
     * @param password   FTP登录密码
     * @param remotePath FTP服务器上的相对路径
     * @param fileName   要下载的文件名
     * @param localPath  下载后保存到本地的路径
     * @return 成功失败
     */
    public static boolean downFile(String url, int port, String username, String password, String remotePath,
                                   String fileName, String localPath) {
        boolean success = false;
        FTPClient ftp = new FTPClient();
        try {
            int reply;
            ftp.connect(url, port);
            // 登录
            // 如果采用默认端口，可以使用ftp.connect(url)的方式直接连接FTP服务器
            ftp.login(username, password);
            reply = ftp.getReplyCode();
            if (!FTPReply.isPositiveCompletion(reply)) {
                ftp.disconnect();
                return success;
            }
            // 转移到FTP服务器目录
            ftp.changeWorkingDirectory(remotePath);
            FTPFile[] fs = ftp.listFiles();
            for (FTPFile ff : fs) {
                String remotFileName = new String(ff.getName().getBytes(StandardCharsets.ISO_8859_1), StandardCharsets.UTF_8);
                if (remotFileName.equals(fileName)) {
                    File localFile = new File(localPath + "/" + remotFileName);
                    OutputStream is = Files.newOutputStream(localFile.toPath());
                    ftp.retrieveFile(ff.getName(), is);
                    is.close();
                }
            }

            ftp.logout();
            success = true;
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            if (ftp.isConnected()) {
                try {
                    ftp.disconnect();
                } catch (IOException ioe) {
                    ioe.printStackTrace();
                }
            }
        }
        return success;
    }

}
